package com.example.springboot_demo.tools;

import com.example.springboot_demo.mappers.AllSchoolMapper;
import com.example.springboot_demo.pojo.AllSchool;
import lombok.SneakyThrows;
import org.apache.commons.lang3.RandomUtils;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.select.Elements;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.*;

@Component
public class JsoupUtils {

    List<String> userAgent = Arrays.asList("Mozilla/4.0(compatible;MSIE7.0;WindowsNT5.1;AvantBrowser)",
            "Mozilla/4.0(compatible;MSIE7.0;WindowsNT5.1;360SE)",
            "Mozilla/4.0(compatible;MSIE7.0;WindowsNT5.1;Trident/4.0;SE2.XMetaSr1.0;SE2.XMetaSr1.0;.NETCLR2.0.50727;SE2.XMetaSr1.0)",
            "Mozilla/4.0(compatible;MSIE7.0;WindowsNT5.1;TheWorld)",
            "Mozilla/4.0(compatible;MSIE7.0;WindowsNT5.1;TencentTraveler4.0)",
            "Opera/9.80(Macintosh;IntelMacOSX10.6.8;U;en)Presto/2.8.131Version/11.11",
            "Mozilla/5.0(WindowsNT6.1;rv:2.0.1)Gecko/20100101Firefox/4.0.1",
            "Mozilla/5.0(compatible;MSIE9.0;WindowsNT6.1;Trident/5.0",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36 OPR/26.0.1656.60",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/30.0.1599.101 Safari/537.36",
            "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E; LBBROWSER)",
            "Mozilla/5.0 (Windows NT 5.1) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.84 Safari/535.11 SE 2.X MetaSr 1.0",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/38.0.2125.122 UBrowser/4.0.3214.0 Safari/537.36",
            "Mozilla/5.0 (X11; U; Linux x86_64; zh-CN; rv:1.9.2.10) Gecko/20100922 Ubuntu/10.10 (maverick) Firefox/3.6.10",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.1 (KHTML, like Gecko) Chrome/14.0.835.163 Safari/535.1",
            "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; AcooBrowser; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
            "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0; Acoo Browser; SLCC1; .NET CLR 2.0.50727; Media Center PC 5.0; .NET CLR 3.0.04506)",
            "Mozilla/4.0 (compatible; MSIE 7.0; AOL 9.5; AOLBuild 4337.35; Windows NT 5.1; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
            "Mozilla/5.0 (Windows; U; MSIE 9.0; Windows NT 9.0; en-US)",
            "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Win64; x64; Trident/5.0; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 2.0.50727; Media Center PC 6.0)",
            "Mozilla/5.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0; WOW64; Trident/4.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 1.0.3705; .NET CLR 1.1.4322)",
            "Mozilla/4.0 (compatible; MSIE 7.0b; Windows NT 5.2; .NET CLR 1.1.4322; .NET CLR 2.0.50727; InfoPath.2; .NET CLR 3.0.04506.30)",
            "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN) AppleWebKit/523.15 (KHTML, like Gecko, Safari/419.3) Arora/0.3 (Change: 287 c9dfb30)",
            "Mozilla/5.0 (X11; U; Linux; en-US) AppleWebKit/527+ (KHTML, like Gecko, Safari/419.3) Arora/0.6",
            "Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.1.2pre) Gecko/20070215 K-Ninja/2.1.1",
            "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9) Gecko/20080705 Firefox/3.0 Kapiko/3.0",
            "Mozilla/5.0 (X11; Linux i686; U;) Gecko/20070322 Kazehakase/0.4.5",
            "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.8) Gecko Fedora/1.9.0.8-1.fc10 Kazehakase/0.5.6",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_3) AppleWebKit/535.20 (KHTML, like Gecko) Chrome/19.0.1036.7 Safari/535.20",
            "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; fr) Presto/2.9.168 Version/11.52");


    @SneakyThrows
    public Map<String, String> getFromBaike(String word) {
        String url = "https://baike.baidu.com/item/%E5%8E%A6%E9%97%A8%E5%8C%BB%E5%AD%A6%E9%99%A2/18633719?fromModule=lemma-qiyi_sense-lemma";
//        String url = String.format("https://baike.baidu.com/item/%s", word);
        HashMap<String, String> heards = new HashMap<>();
        heards.put("User-Agent", userAgent.get(RandomUtils.nextInt(0, userAgent.size() - 1)));
        Document document = Jsoup.connect(url).headers(heards).get();
        Elements names = document.getElementsByClass("basicInfo-item name");
        Elements values = document.getElementsByClass("basicInfo-item value");
        Map<String, String> baikemap = new HashMap<>();
        for (int i = 0; i < names.size();i++){
            baikemap.put(names.get(i).text(), values.get(i).text());
        }
        return baikemap;
    }


    @Autowired
    AllSchoolMapper allSchoolMapper;

    public List<AllSchool> selectAllSchllo(){
        AllSchool xian = allSchoolMapper.selectByPrimaryKey(1558);
        Map<String, String> map = getFromBaike(xian.getSchName());
        xian.setBaikeData(map);
        allSchoolMapper.updateBaike(xian);
//
//        List<AllSchool> all = allSchoolMapper.findAll();
//        for (int i = 525; i < all.size(); i++) {
//            AllSchool item = all.get(i);
//            Map<String, String> map = getFromBaike(item.getSchName());
//            item.setBaikeData(map);
//            allSchoolMapper.updateBaike(item);
//        }
        return null;
    }

    @SneakyThrows
    public Map<String, String> temp() {
        String url = "https://sms24.info/en/messages/OpenAI/";
        Map<String, Integer> map = new HashMap<>();
        for (int i = 1; i <=20 ; i++) {
            String requestUrl = url + i;
            HashMap<String, String> heards = new HashMap<>();
            heards.put("User-Agent", userAgent.get(RandomUtils.nextInt(0, userAgent.size() - 1)));
            Document document = Jsoup.connect(requestUrl).headers(heards).get();
            Elements list = document.getElementsByClass("mb-1");
            for (int j = 0; j < list.size();j++){
                String text = list.get(j).children().get(1).text();
                map.put(text, map.getOrDefault(text, 0) + 1);
            }
        }
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            System.out.println(entry);
        }
        return null;
    }
}
